
import sklearn
import Question2
from sklearn.feature_extraction.text import CountVectorizer

'''
    这个模块调用Question2模块和CountVectorizer，将若干个文档处理为词频矩阵。
'''

def frequency_matrix(file_name_list):
    '''
        此前已经使用Question1模块对所有文件按照日期进行了分类。这个函数首先读取包含所有文件
        路径的文档，然后依次读取这些文档将它们进行分词、处理，并使用CountVectorizer将其转
        变为词频矩阵。
        注：这个模块依赖于Question2模块
    :param:函数接收一个参数，该参数为包含所有待处理文件文件路径的列表。
    :return:返回一个二维矩阵
    '''
    corpus=[]
    for i in file_name_list:
        corpus.append(Question2.sep_words_by_path(i))
    vectorizer = CountVectorizer()
    X = vectorizer.fit_transform(corpus)
    return X

